#include "NvInfer.h"
#include "NvInferPlugin.h"
#include "trt_plugin.hpp"

#include "DpsCostVolumePlugin.hpp"
#include "DpsGeometryVolumePlugin.hpp"
#include "DeformConvPlugin.hpp"
#include "GridSamplePlugin.hpp"

using namespace nvinfer1;
using namespace nvinfer1::plugin;

// namespace nvinfer1
// {
// namespace plugin
// {

// // template <typename CreatorType>
// // void initializePlugin(void* logger, const char* libNamespace)
// // {
// //     PluginCreatorRegistry::getInstance().addPluginCreator<CreatorType>(logger, libNamespace);
// // }

// }
// }
REGISTER_TENSORRT_PLUGIN(DpsCostVolumePluginCreator);
REGISTER_TENSORRT_PLUGIN(DpsGeometryVolumePluginCreator);
REGISTER_TENSORRT_PLUGIN(DeformableConvPluginDynamicCreator);
REGISTER_TENSORRT_PLUGIN(GridSamplePluginDynamicCreator);

extern "C"
{
    bool initLibDsgnInferPlugins() {return true;}
}